我是参照了这两个大佬的 1.dataloader合理设置num_works和batchsize https://blog.csdn.net/qq_41917697/article/details/114663331 2.Pytorch之DataLoader的num_works参数设置 https://blog.csdn.net/qq_41196472/article/details/106393994?utm_source=app&app_version=4.7.1&code=app_1562916241&uLinkId=usr1mkqgl919blen 我的笔记本是CPU 6核12线程的，num_workers=8，设置这个数字是CPU利用率就达到100%了 GPU是6G显存 我的batch_size=16

这样不会爆内存，用第二个大佬的代码测试到最佳num_workers